from houses.models import House
from aijia_project.utils.Error_statuscode import ErrorCode


def generate_housesearch_result(request, data):
    """生成房源搜索API需要的返回结果的辅助函数"""
    # 处理一些在序列化器中未能处理的字段
    for i in data:
        i.pop('facility')
        house = House.objects.get(id=i['house_id'])
        i['user_avata'] = house.user.avatar.url  # 用户头像，是ImageField字段

    # 从查询字符串中获得page参数，如果没有则默认为1
    pages = request.query_params.get('p', 1)

    # 构建符合接口文档要求的返回结果
    result = {
        'data': {
            'houses': data,

            'total_pages': pages
        },
        "errmsg": "请求成功",
        "errno": ErrorCode.OK
    }
    return result
